package com.mlh.binarytree.search;

import com.mlh.binarytree.TreeNode;

import java.util.*;

/**
 * @author 缪林辉
 * @date 2024/3/25 11:08
 * @DESCRIPTION
 */
//第一次写 没想到怎么将不同层的节点装到不同的容器中
//实际上很简单，在最开始遍历的时候，获取队列中的长度len，让len个队列即可
//这样每次队列中全是一层中的节点，不会涉及到上下层

public class BFS {//二叉树层级遍历
    public List<List<Integer>> method1(TreeNode root){
        List<List<Integer>> res=new ArrayList<>();
        if(root==null){return res;}
        Deque<TreeNode> treeNodes = new LinkedList<>();
        treeNodes.add(root);
        while(!treeNodes.isEmpty()){
            int len=treeNodes.size();
            List<Integer>list=new ArrayList<>();
            while (len>0){
                TreeNode node=treeNodes.poll();
                len--;
                 list.add(node.val);
                if(node.left!=null){
                    treeNodes.add(node.left);
                }
                if(node.right!=null){
                    treeNodes.add(node.right);
                }
            }
            res.add(list);
        }
        return res;
    }

    public List<List<Integer>> practice(TreeNode root){
        if(root==null){
            return new ArrayList<>(new ArrayList<>());
        }
        List<List<Integer>>res= new ArrayList<>();
        LinkedList<TreeNode>queue=new LinkedList<>();
        queue.addLast(root);
        while (!queue.isEmpty()){
            ArrayList<Integer> part = new ArrayList<>();
            int len= queue.size();
            while(len>0){
                TreeNode node = queue.pollFirst();
                len--;
                part.add(node.val);
                if(node.left!=null){
                    queue.addLast(node.left);
                }
                if(node.right!=null){
                    queue.addLast(node.right);
                }
            }
            res.add(part);
        }
        return res;
    }

    public List<List<Integer>> practice2(TreeNode root){
        List<List<Integer>> res = new ArrayList<>();
        if(root==null){return res;}
        LinkedList<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        while(!queue.isEmpty()){
            int count= queue.size();
            List<Integer>list=new ArrayList<>();
            for (int i = 0; i < count; i++) {
                TreeNode temp= queue.pollFirst();
                list.add(temp.val);
                if(temp.left!=null){
                    queue.addLast(temp.left);
                }
                if(temp.right!=null){
                    queue.addLast(temp.right);
                }
            }
            res.add(list);
        }
        return res;
    }
}
